package com.project.controller;

import com.project.bean.UserBean;
import com.project.service.IUserService;
import org.apache.catalina.User;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;

@RestController
@RequestMapping("user")
public class UserController {

    @Autowired
    private IUserService service;

    @RequestMapping("login")
    public String login(String name, String pwd, HttpServletRequest req) {
        UserBean user = service.login(name, pwd);
        if (user == null) {
            return "errorInfo";
        }
        req.getSession().setAttribute("loginUser", user);
        return "ok";
    }

    @RequestMapping("getLoginUser")
    public UserBean getLoginUser(HttpServletRequest req) {
        return (UserBean) req.getSession().getAttribute("loginUser");
    }

    @RequestMapping("add")
    public String add(UserBean userBean) {
        service.add(userBean);
        return "ok";
    }
}
